package algorithm7.leetcode.hot100.c3滑动窗口;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class h9_LC438找到字符串中所有字母异位词 {

    /*
        https://leetcode.cn/problems/find-all-anagrams-in-a-string/description/?envType=study-plan-v2&envId=top-100-liked

        438. 找到字符串中所有字母异位词
        给定两个字符串 s 和 p，找到 s 中所有 p 的 异位词 的子串，返回这些子串的起始索引。不考虑答案输出的顺序。
        异位词 指由相同字母重排列形成的字符串（包括相同的字符串）。

        示例 1:
        输入: s = "cbaebabacd", p = "abc"
        输出: [0,6]
        解释:
        起始索引等于 0 的子串是 "cba", 它是 "abc" 的异位词。
        起始索引等于 6 的子串是 "bac", 它是 "abc" 的异位词。

        示例 2:
        输入: s = "abab", p = "ab"
        输出: [0,1,2]
        解释:
        起始索引等于 0 的子串是 "ab", 它是 "ab" 的异位词。
        起始索引等于 1 的子串是 "ba", 它是 "ab" 的异位词。
        起始索引等于 2 的子串是 "ab", 它是 "ab" 的异位词。
     */

    public static void main(String[] args) {
        String s = "cbaebabacd";
        String p = "abc";
        List<Integer> ans = new h9_LC438找到字符串中所有字母异位词().findAnagrams(s, p);
        System.out.println("ans = " + ans);
    }

    public List<Integer> findAnagrams(String s, String p) {
        return solution01(s, p);
    }

    /**
     * 滑动窗口
     * 将p放到一个 int[26] 数组中
     * 使用滑窗遍历s，并将窗口内字串放到一个 int[26] 数组中，判断是否相等，如果相等，l指针为其中一个答案
     * int[26]数组保存的是字符串中某个字母出现的次数
     */
    private List<Integer> solution01(String s, String p) {
        List<Integer> ans = new ArrayList<>();
        if (s == null || p == null || s.length() < p.length()) {
            return ans;
        }
        char[] charArr_s = s.toCharArray();
        char[] charArr_p = p.toCharArray();
        int[] count_s = new int[26];
        int[] count_p = new int[26];

        for (int i = 0; i < charArr_p.length; i++) {
            ++count_p[charArr_p[i] - 'a'];
            ++count_s[charArr_s[i] - 'a'];
        }
        if (Arrays.equals(count_s, count_p)) {
            ans.add(0);
        }

        int windowSize = charArr_p.length;
        for (int r = windowSize; r < charArr_s.length; r++) {
            count_s[charArr_s[r] - 'a']++;
            count_s[charArr_s[r - windowSize] - 'a']--;

            if (Arrays.equals(count_s, count_p)) {
                ans.add(r - windowSize + 1);
            }
        }
        return ans;
    }

}
